package com.lagou.rpc.consumer.proxy;

import com.alibaba.fastjson.JSON;
import com.lagou.rpc.common.RpcRequest;
import com.lagou.rpc.common.RpcResponse;
import com.lagou.rpc.consumer.client.RpcClient;
import org.springframework.stereotype.Component;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.util.*;

/**
 * 客户端代理类--创建代理对象
 * 1、封装request请求对象
 * 2、创建RpcClient对象
 * 3、发送消息
 * 4、返回结果
 */

public class RpcClientProxy {

    private static final List<RpcClient> clients = new ArrayList<>();
    static {
        clients.add(new RpcClient("127.0.0.1", 8899));
        clients.add(new RpcClient("127.0.0.1", 8898));
    }
    public static int index = 0;

   private static  RpcClient getClient() {
        if (index >= clients.size()) {
            index = 0;
        }
        RpcClient rpcClient = clients.get(index);
        index++;
        return rpcClient;
    }

    public static Object createProxy(Class serviceClass){
        return Proxy.newProxyInstance(Thread.currentThread().getContextClassLoader(), new Class[]{serviceClass}, new InvocationHandler() {
            @Override
            public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
                //1、封装request请求对象
                RpcRequest rpcRequest = new RpcRequest();
                rpcRequest.setRequestId(UUID.randomUUID().toString());
                rpcRequest.setClassName(method.getDeclaringClass().getName());
                rpcRequest.setMethodName(method.getName());
                rpcRequest.setParameterTypes(method.getParameterTypes());
                rpcRequest.setParameters(args);
                //2、创建RpcClient对象
               // RpcClient rpcClient = new RpcClient("127.0.0.1",8899);
                RpcClient rpcClient = getClient();
                try {

                    System.out.println("UUID="+rpcRequest.getRequestId());
                    //3、发送消息
                    Object responseMsg = rpcClient.send(JSON.toJSONString(rpcRequest));
                    //把返回信息转化成RpcResponse对象
                    RpcResponse rpcResponse = JSON.parseObject(responseMsg.toString(), RpcResponse.class);
                    if (rpcResponse.getError()!=null){
                        throw new RuntimeException(rpcResponse.getError());
                    }
                    //4、返回结果
                    Object result = rpcResponse.getResult();
                    System.out.println("当前调用的服务端是："+rpcClient.getIp()+":"+rpcClient.getPort());
                    return JSON.parseObject(result.toString(),method.getReturnType());
                }catch (Exception e){
                    throw e;
                }finally {
                    //轮询的时候需要关闭
                    //rpcClient.close();
                }
            }
        });
    }
}
